#%% 定义神经网络
import torch.nn as nn
import torch.nn.functional as F 


class Net(nn.Module):
    """A simple implementation of Deep Neural Network model"""
    def __init__(self):
        super(Net, self).__init__()
        self.input = 470
        self.output = 5
        self.hidden= 300
        self.model = nn.Sequential(
            nn.Linear(self.input, self.hidden),
            #nn.Dropout(0.5),
            nn.ReLU(),
            nn.Linear(self.hidden, self.hidden),
            #nn.Dropout(0.5),
            nn.ReLU(),
            nn.Linear(self.hidden, self.output))
    def forward(self, x):
        return self.model(x)